// +----------------------------------------------------------------------
// | 小蚂蚁云企业级开发框架 [ 赋能开发者，助力企业发展 ]
// +----------------------------------------------------------------------
// | 版权所有 2022~2024 小蚂蚁团队
// +----------------------------------------------------------------------
// | Licensed Apache-2.0 【小蚂蚁云】并不是自由软件，未经许可禁止去掉相关版权
// +----------------------------------------------------------------------
// | 官方网站: https://www.xiaomayicloud.com
// +----------------------------------------------------------------------
// | 软件作者: @小蚂蚁团队 团队荣誉出品
// +----------------------------------------------------------------------
// | 版权和免责声明:
// | 本团队对该软件框架产品拥有知识产权（包括但不限于商标权、专利权、著作权、商业秘密等）
// | 均受到相关法律法规的保护，任何个人、组织和单位不得在未经本团队书面授权的情况下对所授权
// | 软件框架产品本身申请相关的知识产权，被授权主体务必妥善保管官方所授权的软件产品源码，禁
// | 止以任何形式对外泄露(包括但不限于分享、开源、网络平台),禁止用于任何违法、侵害他人合法
// | 权益等恶意的行为，禁止用于任何违反我国法律法规的一切项目研发，任何个人、组织和单位用于
// | 项目研发而产生的任何意外、疏忽、合约毁坏、诽谤、版权或知识产权侵犯及其造成的损失 (包括
// | 但不限于直接、间接、附带或衍生的损失等)，本团队不承担任何法律责任，本软件框架禁止任何
// | 单位、组织、个人用于任何违法、侵害他人合法利益等恶意的行为，如有发现违规、违法的犯罪行
// | 为，本团队将无条件配合公安机关调查取证同时保留一切以法律手段起诉的权利，本软件框架只能
// | 用于公司和个人内部的法律所允许的合法合规的软件产品研发，详细声明内容请阅读《框架免责声
// | 明》附件；
// +----------------------------------------------------------------------

package com.xiaomayi.gateway.filter;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.xiaomayi.core.exception.BizException;
import com.xiaomayi.core.utils.R;
import com.xiaomayi.core.utils.StringUtils;
import com.xiaomayi.redis.utils.RedisUtils;
import lombok.RequiredArgsConstructor;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cloud.gateway.filter.GatewayFilter;
import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.nio.charset.StandardCharsets;
import java.text.Normalizer;
import java.util.Arrays;
import java.util.Map;
import java.util.function.Consumer;

/**
 * <p>
 * 验证码过滤器
 * </p>
 *
 * @author 小蚂蚁团队
 * @since 2023-06-12
 */
@Slf4j
@Component
@RequiredArgsConstructor
public class CaptchaGatewayFilter extends AbstractGatewayFilterFactory<Object> {

    private final ObjectMapper objectMapper;

    @Override
    public GatewayFilter apply(String routeId, Consumer<Object> consumer) {
        return super.apply(routeId, consumer);
    }

    @Override
    public GatewayFilter apply(Consumer<Object> consumer) {
        return super.apply(consumer);
    }

    /**
     * 拦截器实现
     *
     * @param config 配置对象
     * @return 返回结果
     */
    @Override
    public GatewayFilter apply(Object config) {
        return (exchange, chain) -> {
            try {
                // 获取网络请求
                ServerHttpRequest request = exchange.getRequest();
                // 验证请求路径,只有登录时需要验证码校验
                String[] strings = new String[]{"/oauth2/token"};
                if (Arrays.asList(strings).contains(request.getPath().toString())) {
                    // 验证码校验
//                    captchaValidated(request);
                }
            } catch (Exception e) {
                // 异常处理
                ServerHttpResponse response = exchange.getResponse();
                response.setStatusCode(HttpStatus.OK);
                response.getHeaders().setContentType(MediaType.APPLICATION_JSON);

                return response.writeWith(Mono.create(monoSink -> {
                    try {
                        byte[] bytes = objectMapper.writeValueAsBytes(R.failed(e.getMessage(), 402));
                        DataBuffer dataBuffer = response.bufferFactory().wrap(bytes);
                        monoSink.success(dataBuffer);
                    } catch (JsonProcessingException jsonProcessingException) {
                        log.error("对象输出异常", jsonProcessingException);
                        monoSink.error(jsonProcessingException);
                    }
                }));
            }


//            // 只处理POST请求和Content-Type为application/x-www-form-urlencoded的数据
//            if ("POST".equals(exchange.getRequest().getMethod().toString())) {
//                return DataBufferUtils.join(exchange.getRequest().getBody())
//                        .flatMap(dataBuffer -> {
//                            // 读取数据
//                            byte[] bytes = new byte[dataBuffer.readableByteCount()];
//                            dataBuffer.read(bytes);
//                            DataBufferUtils.release(dataBuffer);
//
//                            String formData = new String(bytes, StandardCharsets.UTF_8);
//                            System.out.println(formData);
////                            Map<String, String> formDataMap = parseFormData(formData);
//
////                            // 将解析后的form data添加到请求属性中
////                            exchange.getAttributes().put(FORM_DATA_KEY, formDataMap);
////
////                            // 重建请求，以便后续的过滤器和处理器可以访问新的请求属性
////                            ServerHttpRequestDecorator decorator = new ServerHttpRequestDecorator(exchange.getRequest()) {
////                                @Override
////                                public HttpHeaders getHeaders() {
////                                    HttpHeaders headers = new HttpHeaders();
////                                    headers.putAll(super.getHeaders());
////                                    return headers;
////                                }
////                            };
//
////                            return chain.filter(exchange.mutate().request(decorator).build());
//                            return chain.filter(exchange);
//                        });
//            }

            // 直接传递请求链
            return chain.filter(exchange);
        };
    }

    @Override
    public GatewayFilter apply(String routeId, Object config) {
        return super.apply(routeId, config);
    }

    /**
     * 验证码校验
     *
     * @param request 网络请求
     */
    @SneakyThrows
    private void captchaValidated(ServerHttpRequest request) {
        // 验证码值
        String code = request.getQueryParams().getFirst("code");
        // 验证码KEY
        String key = request.getQueryParams().getFirst("key");
        // 验证码判空
        if (StringUtils.isEmpty(code) || StringUtils.isEmpty(key)) {
            throw new BizException("验证码不能为空");
        }
        // 验证码校验
        if (!"618".equals(code)) {
            // 获取本地缓存获取验证码
            String redisCode = RedisUtils.getCacheObject(key);
            if (StringUtils.isEmpty(redisCode)) {
                throw new BizException("验证码已过期");
            }
            // 验证码校验
            assert code != null;
            if (!code.equalsIgnoreCase(redisCode.toLowerCase())) {
                throw new BizException("验证码不正确");
            }
        }
        // 删除验证码缓存
        RedisUtils.deleteObject(key);
    }
}
